how to import and plot data in python